import pymysql
from itertools import chain


def dbexec(sql):
    db = pymysql.connect(host="localhost", user="root", password="root", database="flaskbigscreen")

    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    # 执行SQL语句
    cursor.execute(sql)
    # 获取所有记录列表
    results = cursor.fetchall()
    # 关闭数据库连接
    db.close()
    return results


def dbexeclist(sql):
    db = pymysql.connect(host="localhost", user="root", password="root", database="flaskbigscreen")

    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    # 执行SQL语句
    cursor.execute(sql)
    # 获取所有记录列表
    results = list(chain.from_iterable(cursor.fetchall()))
    # 关闭数据库连接
    db.close()
    return results


